'''
pt1性能测试视频的入口
存放需要处理的视频路径
存放需要填入的xlsx路径
'''
import os
from multiprocessing import Process

from pt1.process import process
#旧视频的路径

raw_video = [
    '/Users/zltest/Desktop/wired_720.mp4',  #1
    '/Users/zltest/Desktop/wired_1080.mp4',  #2
    # /Users/zltest/Desktop/wired_480.mp4'  #3
    #4,5,6
]

# 定义写入的xlsx路径# 要写入的xlsx文件路径，请注意，‘-’是无法进行写入的，需要手动把文件名里的‘-’更改成‘_’后执行完脚本再改回‘-’
xlsx_path = '/Users/zltest/Desktop/PT1_0312.xlsx'

'''
给定的ROI (0, 720, 1920, 360) 表示的是图像中一个特定区域的位置和大小，其中：

x=0 表示该区域从水平方向最左边开始。
y=720 表示该区域从垂直方向上距离图像顶部720像素处开始。
width=1920 表示该区域的宽度为1920像素，即覆盖整个图像的宽度。
height=360 表示该区域的高度为360像素。
'''
摄像头的roi = (0, 710, 960, 370)  # (x, y, width, height)
绿色文字的roi = (480, 0, 960, 540)

def process_video(address, xlsx_path, roi, word_roi):
    """封装process函数，用于不同视频的处理"""
    process(address=address, xlsx_path=xlsx_path, roi=roi, word_roi=word_roi)

if __name__ == '__main__':
    processes = []
    for video in raw_video:
        p = Process(target=process_video, args=(video, xlsx_path, 摄像头的roi, 绿色文字的roi))
        p.start()
        processes.append(p)

    # 等待所有进程完成
    for p in processes:
        p.join()

    print('All videos have been processed.')